import train
import generate

def main():
    # 训练模型
    train.train()
    # 生成图片
    generate.generate()


if __name__ == '__main__':
    main()